package com.haust.leyou.service;

import com.haust.leyou.mapper.UserMapper;
import com.haust.leyou.pojo.User;
import com.haust.leyou.utils.CodecUtils;
import com.haust.leyou.utils.NumberUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.core.AmqpTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @Auther: csp1999
 * @Date: 2020/12/17/21:13
 * @Description:
 */
@Service
public class UserService {

    @Autowired
    private UserMapper userMapper;

    @Autowired
    private AmqpTemplate amqpTemplate;

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    // 用户短信验证码 key 的前缀
    private static final String KEY_PREFIX = "user:verify:";

    /**
     * 校验用户数据是否可用（用户是否存在）
     *
     * @param data 要校验的数据
     * @param type 要校验的数据类型：1，用户名；2，手机；
     * @return true：可用/false：不可用
     */
    public Boolean checkUser(String data, Integer type) {

        User record = new User();
        switch (type) {
            case 1:
                // type == 1 用户名
//                System.out.println(data);
                record.setUsername(data);
                break;
            case 2:
                // type == 2 手机
//                System.out.println(data);
                record.setPhone(data);
                break;
            default:
                return null;
        }

        return this.userMapper.selectCount(record) == 0;
    }

    /**
     * 发送短信验证码
     *
     * @param phone
     */
    public void sendVerifyCode(String phone) {

        // 前置判断
        if (StringUtils.isBlank(phone)) {
            return;
        }

        // 生成6位数验证码
        String code = NumberUtils.generateCode(6);
        Map<String, String> map = new HashMap<>();
        map.put("phone", phone);
        map.put("code", code);

        // 发送消息到rabbitmq ---> 消息消费者收到消息后发送短信验证码
        amqpTemplate.convertAndSend("LEYOU.SMS.EXCHANGE", "verifycode.sms", map);

        // 把验证码存入到redis中,过期时间5分钟
        stringRedisTemplate.opsForValue().set(KEY_PREFIX + phone, code, 5, TimeUnit.MINUTES);
    }

    /**
     * 用户注册
     *
     * @param user 用户
     * @param code 验证码
     */
    public void register(User user, String code) {

        // 查询redis 中的验证码
        String redisCode = stringRedisTemplate.opsForValue().get(KEY_PREFIX + user.getPhone());

        // 1.校验验证码
        if (!StringUtils.equals(code, redisCode)) {
            // 用户传入的验证码和redis 缓存的验证码不符
            return;
        }

        // 2.生成盐值
        String salt = CodecUtils.generateSalt();
        user.setSalt(salt);

        // 3.加盐加密
        String password = CodecUtils.md5Hex(user.getPassword(), salt);
        user.setPassword(password);

        // 4.新增用户
        user.setId(null);
        user.setCreated(new Date());
        userMapper.insertSelective(user);

        // 5.注册成功后删除redis 中的验证码
        stringRedisTemplate.delete(KEY_PREFIX + user.getPhone());
    }

    /**
     * 根据用户名和密码查询用户信息
     *
     * @param username
     * @param password
     * @return
     */
    public User queryUser(String username, String password) {

        User record = new User();
        record.setUsername(username);

        // 根据用户名查询用户
        User user = userMapper.selectOne(record);

        // 判断user是否为空
        if (user == null) {
            return null;
        }

        // 获取盐值，并对用户输入的密码加盐
        password = CodecUtils.md5Hex(password, user.getSalt());

        // 和数据库中加密的密码进行比较
        if (StringUtils.equals(password,user.getPassword())){
            return user;
        }

        return null;
    }
}
